import pygame
from planeSprites import *
# 游戏屏幕大小
SCREEN_RECT = pygame.Rect(0, 0, 480, 700)
#刷新帧率
FRAME_PER_SEC = 60
# 敌机的定时器事件常量
CREATE_ENEMY_EVENT = pygame.USEREVENT
#英雄发射子弹事件
HERO_FIRE_EVENT = pygame.USEREVENT + 1

class PlaneGame(object):
    """飞机大战主游戏"""
    def __init__(self):
        print("游戏初始化")
        # 1. 创建游戏的窗口
        self.screen = pygame.display.set_mode(SCREEN_RECT.size)
        # 2. 创建游戏的时钟
        self.clock = pygame.time.Clock()
        # 3. 调用私有方法，精灵和精灵组的创建
        self.__create_sprites()
        # 4. 设置定时器事件 - 每秒创建一架敌机
        pygame.time.set_timer(CREATE_ENEMY_EVENT, 1000)
        # 5. 设置定时器事件 - 每0.5秒发射一次子弹
        pygame.time.set_timer(HERO_FIRE_EVENT, 300)

    def __create_sprites(self):
        # 创建背景精灵和精灵组
        bg1 = BackGroundSprite()
        bg2 = BackGroundSprite(True)
        self.back_group = pygame.sprite.Group(bg1, bg2)
        # 敌机组
        self.enemy_group = pygame.sprite.Group()
        # 英雄组
        self.hero = Hero()
        self.hero_group = pygame.sprite.Group(self.hero)

    def start_game(self):
        print("开始游戏...")
        while True:
            # 1. 设置刷新帧率
            self.clock.tick(FRAME_PER_SEC)
            # 2. 事件监听
            self.__event_handler()
            # 3. 碰撞检测
            self.__check_collide()
            # 4. 更新精灵组
            self.__update_sprites()
            # 5. 更新屏幕显示
            pygame.display.update()

    @staticmethod
    def __game_over():
        print("游戏结束")
        pygame.quit()
        exit()

    def __event_handler(self):
        """事件监听"""
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                PlaneGame.__game_over()
            elif event.type == CREATE_ENEMY_EVENT:
                print("敌机出场...")
                #每1秒把敌机添加到精灵组当中
                self.enemy_group.add(Enemy())
            elif event.type == HERO_FIRE_EVENT:
                #发射子弹
                self.hero.fire()

        # 返回所有按键的元组，如果某个键被按下，对应的值会是1
        keys_pressed = pygame.key.get_pressed()
        # 判断是否按下了方向键
        if keys_pressed[pygame.K_RIGHT]:
            self.hero.speed = 5
        elif keys_pressed[pygame.K_LEFT]:
            self.hero.speed = -5
        else:
            self.hero.speed = 0

    def __check_collide(self):
        """碰撞检测"""
        # 1. 子弹摧毁敌机
        pygame.sprite.groupcollide(self.hero.bulletsGroup, self.enemy_group, True, True)
        # 2. 敌机撞毁英雄
        enemies = pygame.sprite.spritecollide(self.hero, self.enemy_group, True)
        # 判断列表时候有内容
        if len(enemies) > 0:
            # 让英雄牺牲
            self.hero.kill()
            #游戏结束
            PlaneGame.__game_over()

    def __update_sprites(self):
        """更新精灵组"""
        self.back_group.update()
        self.back_group.draw(self.screen)
        #更新敌机
        self.enemy_group.update()
        self.enemy_group.draw(self.screen)
        #绘制英雄
        self.hero_group.update()
        self.hero_group.draw(self.screen)
        #绘制子弹
        self.hero.bulletsGroup.update()
        self.hero.bulletsGroup.draw(self.screen)


if __name__ == '__main__':
    # 创建游戏对象
    game = PlaneGame()
    # 开始游戏
    game.start_game()


